package com.ssyan.datastructure.linear;

import java.util.Iterator;

/*
    单向链表：
        1.基础API实现
        2.遍历实现
        3.容量可变实现
        4.时间复杂度：增删改查O(n)
 */
public class LinkList<T> implements Iterable<T>{

    //记录头结点
    private Node head;
    //记录链表的长度
    private int N;

    //成员内部类：结点类
    private class Node {
        //存储数据
        T item;
        //下一个结点
        Node next;

        public Node(T item,Node next) {
            this.item = item;
            this.next = next;
        }
    }

    //构造方法
    public LinkList() {
        //初始化结点
        this.head = new Node(null,null);
        this.N = 0;
    }

    //将一个线性表置为空表
    public void clear() {
        head.next=null;
        this.N=0;
    }

    //判断当前线性表是否为空表
    public boolean isEmpty() {

        return N==0;
    }

    //获取线性表的长度
    public int length() {
        return N;
    }

    //获取指定位置的元素
    public T get(int i) {

        //通过循环，从头结点开始往后找，依次找i次，就可以找到对应的元素
        Node n = head.next;
        for (int index = 0; index < i; index++) {
            n = n.next;
        }

        return n.item;
    }

    //向线性表中添加元素t
    public void insert(T t) {

        //找到当前最后一个结点
        Node n = head;
        while (n.next != null) {
            n = n.next;
        }

        //创建新结点，保存元素t
        Node newNode = new Node(t, null);

        //让当前最后一个结点指向新结点
        n.next=newNode;

        //元素个数+1
        N++;
    }

    //向指定位置i处，插入元素t
    public void insert(int i,T t) {

        //找到i位置前一个结点
        Node pre = head;
        for (int index = 0; index <= i - 1; index++) {
            pre = pre.next;
        }
        //找到i位置的结点
        Node curr = pre.next;

        //创建新结点，并且新结点需要指向原来i位置的结点
        Node newNode = new Node(t, curr);

        //原来i位置的前一个结点指向新结点即可
        pre.next = newNode;

        //元素的个数+1
        N++;
    }

    //删除指定位置i处的元素，并返回该元素
    public T remove(int i) {
        //找到i位置的前一个结点
        Node pre = head;
        for (int index = 0; index <= i-1; index++) {
            pre = pre.next;
        }
        //要找到i位置的结点
        Node curr = pre.next;

        //找到i位置的下一个结点
        Node nextNode = curr.next;

        //前一个结点指向下一个结点
        pre.next = nextNode;

        //元素个数-1
        N--;

        return curr.item;
    }

    //查找t元素第一次出现的位置
    public int indexOf(T t) {
        //从头结点开始，依次找到每一个结点，取出item，和t比较，如果相同，就找到了
        Node n = head;
        for (int i=0;n.next!=null;i++) {
            n=n.next;
            if (t.equals(n.item)) {
                return i;
            }
        }
        return -1;
    }

    /*
        实现遍历
     */
    @Override
    public Iterator<T> iterator() {
        return new LIterator();
    }

    private class LIterator implements Iterator<T> {
        private Node n;

        public LIterator() {
            this.n = head;
        }

        @Override
        public boolean hasNext() {
            return n.next != null;
        }

        @Override
        public T next() {
            n = n.next;
            return n.item;
        }
    }


}
